from pydantic import BaseModel, EmailStr, Field
from datetime import datetime
from typing import Optional

# 基础用户模型（共享属性）
class UserBase(BaseModel):
    username: str
    email: EmailStr
    full_name: Optional[str] = None

# 创建用户时使用（不包含ID和时间戳）
class UserCreate(UserBase):
    password: str = Field(..., min_length=8)  # 密码至少8位

# 更新用户时使用（所有字段可选）
class UserUpdate(BaseModel):
    email: Optional[EmailStr] = None
    full_name: Optional[str] = None
    password: Optional[str] = Field(None, min_length=8)

# 数据库模型映射（包含ID和时间戳）
class UserInDBBase(UserBase):
    id: int
    is_active: bool
    is_admin: bool
    created_at: datetime
    updated_at: Optional[datetime] = None

    class Config:
        from_attributes = True  # 支持ORM模型转换

# 返回给客户端的用户模型
class User(UserInDBBase):
    pass

# 数据库中的用户模型（包含密码哈希）
class UserInDB(UserInDBBase):
    hashed_password: str

# 登录请求模型
class UserLogin(BaseModel):
    username: str
    password: str

# 令牌模型
class Token(BaseModel):
    access_token: str
    token_type: str

# 令牌数据模型
class TokenData(BaseModel):
    user_id: Optional[int] = None
